﻿using System.ComponentModel;
using NewLife.Data;
using NewLife.Log;
using XCode;

namespace NewLife.Cube.Entity;

/// <summary>安全访问动作类型</summary>
public enum AccessActionKinds
{
    /// <summary>放行</summary>
    [Description("放行")]
    Pass = 1,

    /// <summary>拦截</summary>
    [Description("拦截")]
    Block = 2,

    /// <summary>限流</summary>
    [Description("限流")]
    Limit = 3,
}

/// <summary>限流维度</summary>
public enum LimitDimensions
{
    /// <summary>按IP</summary>
    [Description("按IP")]
    IP = 1,

    /// <summary>按用户</summary>
    [Description("按用户")]
    User = 2,
}

public partial class AccessRule : Entity<AccessRule>
{
    #region 对象操作
    static AccessRule()
    {
        // 累加字段，生成 Update xx Set Count=Count+1234 Where xxx
        //var df = Meta.Factory.AdditionalFields;
        //df.Add(nameof(Priority));

        // 过滤器 UserModule、TimeModule、IPModule
        Meta.Modules.Add(new UserModule { AllowEmpty = false });
        Meta.Modules.Add<TimeModule>();
        Meta.Modules.Add(new IPModule { AllowEmpty = false });

        // 实体缓存
        // var ec = Meta.Cache;
        // ec.Expire = 60;

        // 单对象缓存
        var sc = Meta.SingleCache;
        // sc.Expire = 60;
        sc.FindSlaveKeyMethod = k => Find(_.Name == k);
        sc.GetSlaveKeyMethod = e => e.Name;
    }

    /// <summary>验证并修补数据，返回验证结果，或者通过抛出异常的方式提示验证失败。</summary>
    /// <param name="method">添删改方法</param>
    public override Boolean Valid(DataMethod method)
    {
        //if (method == DataMethod.Delete) return true;
        // 如果没有脏数据，则不需要进行任何处理
        if (!HasDirty) return true;

        // 这里验证参数范围，建议抛出参数异常，指定参数名，前端用户界面可以捕获参数异常并聚焦到对应的参数输入框
        if (Name.IsNullOrEmpty()) throw new ArgumentNullException(nameof(Name), "名称不能为空！");

        // 建议先调用基类方法，基类方法会做一些统一处理
        if (!base.Valid(method)) return false;

        // 在新插入数据或者修改了指定字段时进行修正

        // 处理当前已登录用户信息，可以由UserModule过滤器代劳
        /*var user = ManageProvider.User;
        if (user != null)
        {
            if (method == DataMethod.Insert && !Dirtys[nameof(CreateUserID)]) CreateUserID = user.ID;
            if (!Dirtys[nameof(UpdateUserID)]) UpdateUserID = user.ID;
        }*/
        //if (method == DataMethod.Insert && !Dirtys[nameof(CreateTime)]) CreateTime = DateTime.Now;
        //if (!Dirtys[nameof(UpdateTime)]) UpdateTime = DateTime.Now;
        //if (method == DataMethod.Insert && !Dirtys[nameof(CreateIP)]) CreateIP = ManageProvider.UserHost;
        //if (!Dirtys[nameof(UpdateIP)]) UpdateIP = ManageProvider.UserHost;

        // 检查唯一索引
        // CheckExist(method == DataMethod.Insert, nameof(Name));

        return true;
    }

    /// <summary>首次连接数据库时初始化数据，仅用于实体类重载，用户不应该调用该方法</summary>
    [EditorBrowsable(EditorBrowsableState.Never)]
    protected override void InitData()
    {
        // InitData一般用于当数据表没有数据时添加一些默认数据，该实体类的任何第一次数据库操作都会触发该方法，默认异步调用
        if (Meta.Session.Count > 0) return;

        if (XTrace.Debug) XTrace.WriteLine("开始初始化AccessRule[访问规则]数据……");

        var entity = new AccessRule
        {
            Name = "封禁IP",
            Enable = false,
            Url = "*",
            UserAgent = "*",
            IP = "10.0.*",
            ActionKind = AccessActionKinds.Block,
            BlockCode = 302,
            BlockContent = "https://newlifex.com",
            Remark = "指定IP段访问时，直接跳转目标Url",
        };
        entity.Insert();

        entity = new AccessRule
        {
            Name = "封禁爬虫",
            Enable = false,
            Url = "!/Admin/*,!/Cube/*",
            UserAgent = "*bot*,*spider*",
            IP = "",
            ActionKind = AccessActionKinds.Block,
            BlockCode = 302,
            BlockContent = "https://newlifex.com",
            Remark = "蜘蛛访问Admin和Cube以外路径时，直接跳转目标Url",
        };
        entity.Insert();

        entity = new AccessRule
        {
            Name = "IP访问太快",
            Enable = false,
            Url = "!*.js,*.css,*.ico,*.jpg,*.png",
            UserAgent = "",
            IP = "",
            ActionKind = AccessActionKinds.Limit,
            BlockCode = 200,
            BlockContent = "<h1>访问太频繁啦，坐下来喝杯咖啡吧！</h1>",
            LimitDimension = LimitDimensions.IP,
            LimitCycle = 60,
            LimitTimes = 100,
            Remark = "单个IP访问过快时（60秒超100次），给出友好提示！",
        };
        entity.Insert();

        entity = new AccessRule
        {
            Name = "内网优先",
            Enable = false,
            Priority = 999,
            IP = "192.*",
            ActionKind = AccessActionKinds.Pass,
            Remark = "内网可信来源访问，直接放行！",
        };
        entity.Insert();

        if (XTrace.Debug) XTrace.WriteLine("完成初始化AccessRule[访问规则]数据！");
    }
    #endregion

    #region 扩展属性
    #endregion

    #region 扩展查询
    /// <summary>根据编号查找</summary>
    /// <param name="id">编号</param>
    /// <returns>实体对象</returns>
    public static AccessRule FindById(Int32 id)
    {
        if (id <= 0) return null;

        // 实体缓存
        if (Meta.Session.Count < 1000) return Meta.Cache.Find(e => e.Id == id);

        // 单对象缓存
        return Meta.SingleCache[id];

        //return Find(_.Id == id);
    }

    /// <summary>根据名称查找</summary>
    /// <param name="name">名称</param>
    /// <returns>实体对象</returns>
    public static AccessRule FindByName(String name)
    {
        if (name.IsNullOrEmpty()) return null;

        // 实体缓存
        if (Meta.Session.Count < 1000) return Meta.Cache.Find(e => e.Name.EqualIgnoreCase(name));

        // 单对象缓存
        //return Meta.SingleCache.GetItemWithSlaveKey(name) as AccessRule;

        return Find(_.Name == name);
    }
    #endregion

    #region 高级查询
    /// <summary>高级查询</summary>
    /// <param name="name">名称</param>
    /// <param name="start">更新时间开始</param>
    /// <param name="end">更新时间结束</param>
    /// <param name="key">关键字</param>
    /// <param name="page">分页参数信息。可携带统计和数据权限扩展查询等信息</param>
    /// <returns>实体列表</returns>
    public static IList<AccessRule> Search(String name, DateTime start, DateTime end, String key, PageParameter page)
    {
        var exp = new WhereExpression();

        if (!name.IsNullOrEmpty()) exp &= _.Name == name;
        exp &= _.UpdateTime.Between(start, end);
        if (!key.IsNullOrEmpty()) exp &= _.Name.Contains(key) | _.Url.Contains(key) | _.UserAgent.Contains(key) | _.IP.Contains(key) | _.LoginedUser.Contains(key) | _.BlockContent.Contains(key) | _.CreateIP.Contains(key) | _.UpdateIP.Contains(key) | _.Remark.Contains(key);

        return FindAll(exp, page);
    }

    // Select Count(Id) as Id,Category From AccessRule Where CreateTime>'2020-01-24 00:00:00' Group By Category Order By Id Desc limit 20
    //static readonly FieldCache<AccessRule> _CategoryCache = new FieldCache<AccessRule>(nameof(Category))
    //{
    //Where = _.CreateTime > DateTime.Today.AddDays(-30) & Expression.Empty
    //};

    ///// <summary>获取类别列表，字段缓存10分钟，分组统计数据最多的前20种，用于魔方前台下拉选择</summary>
    ///// <returns></returns>
    //public static IDictionary<String, String> GetCategoryList() => _CategoryCache.FindAllName();
    #endregion

    #region 业务操作
    #endregion
}
